package com.zxy.allSearcher.service.impl;

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.zxy.allSearcher.common.ErrorCode;
import com.zxy.allSearcher.exception.BusinessException;
import com.zxy.allSearcher.model.entity.Picture;
import com.zxy.allSearcher.service.PictureService;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;
import org.springframework.stereotype.Service;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

/**
 * 图片服务实现
 * 幡然醒悟，我不是还要github提供的数据包吗！！！
 */
@Service
public class PictureServiceImpl implements PictureService {
    @Override
    public Page<Picture> searchPicture(String searchText, long pageNum, long pageSize) {
        // 计算当前页面
        long current = (pageNum - 1) * pageSize;
        // 1、数据获取
        String url = String.format("https://www.bing.com/images/search?q=%s&first=%s", searchText, current);
        Document doc = null;
        try {
            doc = Jsoup.connect(url).get();
        } catch (IOException e) {
            throw new BusinessException(ErrorCode.SYSTEM_ERROR, "数据获取异常");
        }
        Elements elements = doc.select(".iuscp.isv");
        // 2、数据转换，注意有些图片数据是压缩，懒加载，缩略图的形式，是无法获取完整的图源信息的（得爬仔细）
        List<Picture> pictures = new ArrayList<>();
        // 深度爬取
        for (Element element : elements) {
            // 图片url
            String src = element.select(".mimg").get(0).attr("src");
            System.out.println(src);
            // 标题
            String title = element.select(".inflnk").get(0).attr("aria-label");
            System.out.println(title);
            // 数据封装
            Picture picture = new Picture();
            picture.setUrl(src);
            picture.setTitle(title);

            pictures.add(picture);

            // 根据分页大小，终止转换，感觉在爬取时就应该约定好，还可以提升性能+一致性
            if (pictures.size() >= pageSize) {
                break;
            }
        }
        // 3、分页展示，由于无法估计爬出的数据总数，无法返回
        Page<Picture> picturePage = new Page<>(pageNum, pageSize);
        picturePage.setRecords(pictures);
        return picturePage;
    }
}




